package com.kevinkk.hashmap;

import java.util.Set;
import java.util.HashSet;

/**
 * 编写一个算法来判断一个数 n 是不是快乐数。「快乐数」 定义为：
 * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
 * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
 * 如果这个过程 结果为 1，那么这个数就是快乐数。
 * 如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
 */

/**
 * 这道题的关键就在于，在计算 快乐数 的过程中可能会出现 “无限循环” 的现象
 * 因此用一个 Set 去记录计算过程，如果 Set 中有重复数字出现，那么肯定不是快乐数
 */

public class HappyNumber {
    class Solution {
        public boolean isHappy(int n) {
            Set<Integer> set = new HashSet<>();
            while (n != 1 && !set.contains(n)) {
                set.add(n);
                n = getHappy(n);
            }
            return n == 1;
        }

        private int getHappy(int n) {
            int res = 0;
            while (n != 0) {
                int digit = n % 10;
                res += digit * digit;
                n /= 10;
            }
            return res;
        }
    }
}
